import tensorflow as tf

input_shape = (2, 4, 5, 3)
x = tf.random.normal(input_shape)
tf.print(x)
y = tf.keras.layers.GlobalAveragePooling2D()(x)
tf.print(y)

y00 = tf.reduce_mean(x[0, :, :, 0])
tf.print(y00)
y01 = tf.reduce_mean(x[0, :, :, 1])
tf.print(y01)
